package com.changgou.file.controller;

import com.changgou.entity.Result;
import com.changgou.entity.StatusCode;
import com.changgou.file.util.FastDFSClient;
import com.changgou.file.util.FastDFSFile;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;

@RestController
@RequestMapping("/file")
public class FileController {

    private Logger logger = LoggerFactory.getLogger(FastDFSClient.class);

    @PostMapping("/upload")
    public Result upload(@RequestParam("file") MultipartFile file) {
        String filename = null;
        try {
            filename = file.getOriginalFilename();
            String extName = filename.substring(filename.lastIndexOf("."));
            byte[] fileContent = file.getBytes();
            FastDFSFile fastDFSFile = new FastDFSFile(filename, fileContent, extName);
            String[] uploadResult = FastDFSClient.upload(fastDFSFile);
            if (null != uploadResult && uploadResult.length == 2) {
                String groupName = uploadResult[0];
                String remoteFilePath = uploadResult[1];
                String url = FastDFSClient.getTrackerUrl() + groupName + "/" + remoteFilePath;
                logger.info("上传成功, url:{}", url);
                return new Result(true, StatusCode.OK, "上传成功", url);
            } else {
                logger.error("上传失败, fileName:{}", filename);
                return new Result(false, StatusCode.ERROR, "上传失败");
            }

        } catch (IOException e) {
            e.printStackTrace();
            logger.error("上传失败, fileName:{}", filename);
            return new Result(false, StatusCode.ERROR, "上传失败");
        }

    }

}
